"""
Copyright (c) 2022 Huawei Technologies Co.,Ltd.

openGauss is licensed under Mulan PSL v2.
You can use this software according to the terms and conditions of the Mulan PSL v2.
You may obtain a copy of Mulan PSL v2 at:

          http://license.coscl.org.cn/MulanPSL2

THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
See the Mulan PSL v2 for more details.
"""
"""
Case Type   : sqlines工具
Case Name   : sqlines工具-s参数指定oracle，创建视图，查看转换情况
Create At   : 2022/01/24
@zou_jialiang0501115624
Description :
    1、sqlines安装
    2、创建测试脚本
    3、sqlines -s=oracle -t=opengauss -in={self.in_file} -out={self.out_file}
     -log={self.log_file}
    4、环境清理：卸载sqlines、删除对应文件
Expect      :
    1、sqlines安装成功
    2、脚本创建成功
    3、转换成功，删除CREATE OR REPLACE与VIEW之间的参数，删除sharing选项
    4、环境清理成功
History     :
    modified: 2023-04-19 by @daiguatutu 优化unzip解压命令，避免命令执行超时失败
"""

import os
import unittest

from yat.test import Node
from yat.test import macro
from testcase.utils.Logger import Logger


class SqlinesTest(unittest.TestCase):

    def setUp(self):
        self.log = Logger()
        self.log.info(f'----- {os.path.basename(__file__)} start-----')
        self.root = Node('PrimaryRoot')
        self.in_file = f'{os.path.basename(__file__)[:-3]}.sql'
        self.out_file = f'{os.path.basename(__file__)[:-3]}_out.sql'
        self.log_file = f'{os.path.basename(__file__)[:-3]}.log'
        self.file_name = 'openGauss-tools-sqlines-master.zip'
        self.file_path = '/root/sqlines_034/'

    def test_main(self):
        step_txt = '----step1:安装sqlines，expect: sqlines安装成功----'
        self.log.info(step_txt)
        self.log.info('-------下载软件包--------')
        file_download = os.path.join(macro.FTP_PLUGINS_PATH,
                                     'sqlines', self.file_name)
        cmd = f'mkdir -p {self.file_path};' \
            f'wget -P {self.file_path} {file_download}'
        self.log.info(cmd)
        self.root.sh(cmd)
        cmd2 = f'ls {self.file_path}{self.file_name}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertEqual(self.file_path + self.file_name, res,
                         '文件下载失败：' + step_txt)

        self.log.info('-------解压软件包--------')
        cmd3 = f'timeout 30s unzip -o -d {self.file_path} ' \
            f'{self.file_path}{self.file_name}'
        self.log.info(cmd3)
        res = self.root.sh(cmd3).result()
        self.log.info(res)
        self.unzip_dir = self.file_path + self.file_name.replace('.zip', '')
        cmd4 = f'ls -d {self.unzip_dir}'
        self.log.info(cmd4)
        res = self.root.sh(cmd4).result()
        self.log.info(res)
        self.assertEqual(self.unzip_dir, res, '文件解压失败：' + step_txt)

        self.log.info('-------安装sqlines--------')
        cmd = f'cd {self.unzip_dir};sh build.sh -i;pwd'
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertIn('Install Sqlines successfully', res,
                      'Sqlines安装失败：' + step_txt)
        tmp_path = res.strip().splitlines()[-1].strip()
        mid_path = os.path.join(tmp_path, 'bin')
        file_name = 'sqlines'
        check_file = os.path.join(mid_path, file_name)
        cmd = f"ls -l {check_file}"
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertEqual(res[3], 'x', '文件没有执行权限' + step_txt)
        self.log.info('-----配置sqlines环境变量-----')
        self.source_path = f"export PATH={mid_path}:$PATH"

        step_txt = '----step2:创建测试脚本，expect: 脚本创建成功----'
        self.log.info(step_txt)
        cmd1 = f"""cat >{self.in_file} <<EOF
        create or replace force editioning view 
          v_Opengauss_Function_Sqlines_Case0034_1 
          sharing=none as select * from emp;
        create or replace no force editionable editioning view 
          v_Opengauss_Function_Sqlines_Case0034_2 
          sharing=metadata as select * from emp;
        create or replace force noneditionable view
          v_Opengauss_Function_Sqlines_Case0034_3 
          sharing=data as select * from emp;
        create or replace view v_Opengauss_Function_Sqlines_Case0034_4 
          sharing=extended data as select * from emp;
        """
        self.log.info(cmd1)
        self.root.sh(cmd1)
        cmd2 = f'ls {self.in_file}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertEqual(res, f'{self.in_file}', '执行失败' + step_txt)

        step_txt = '----step3:执行sqlines命令，expect: 转换成功，' \
                   '删除CREATE OR REPLACE与VIEW之间的参数，删除sharing选项'
        self.log.info(step_txt)
        cmd = f"{self.source_path};" \
            f"sqlines -s=oracle -t=opengauss -in={self.in_file}" \
            f" -out={self.out_file} -log={self.log_file}"
        self.log.info(cmd)
        res = self.root.sh(cmd).result()
        self.log.info(res)
        self.assertNotIn('error', res, '执行失败：')
        cmd2 = f'cat {self.out_file}'
        self.log.info(cmd2)
        res = self.root.sh(cmd2).result()
        self.log.info(res)
        self.assertTrue(len(res.strip().splitlines()) >= 11, '行数不到11行')

        check1 = res.strip().splitlines()[0].strip()
        check2 = res.strip().splitlines()[2].strip()
        check3 = res.strip().splitlines()[3].strip()
        check4 = res.strip().splitlines()[5].strip()
        check5 = res.strip().splitlines()[6].strip()
        check6 = res.strip().splitlines()[8].strip()
        check7 = res.strip().splitlines()[10].strip()
        self.assertNotIn('force editioning', check1, '执行失败：' + step_txt)
        self.assertNotIn('sharing=none', check2, '执行失败：' + step_txt)
        self.assertNotIn('no force editionable editioning',
                         check3, '执行失败：' + step_txt)
        self.assertNotIn('sharing=metadata', check4, '执行失败：' + step_txt)
        self.assertNotIn('force noneditionable', check5, '执行失败：' + step_txt)
        self.assertNotIn('sharing=data', check6, '执行失败：' + step_txt)
        self.assertNotIn('sharing=extended', check7, '执行失败：' + step_txt)

    def tearDown(self):
        step_txt = '----step4:环境清理，expect: 环境清理成功----'
        self.log.info(step_txt)
        self.log.info('------卸载sqlines------')
        cmd1 = f'cd {self.unzip_dir};sh build.sh -m'
        self.log.info(cmd1)
        res1 = self.root.sh(cmd1).result()
        self.log.info(res1)
        self.log.info('------删除相关文件------')
        cmd2 = f"rm -rf {self.file_path} {self.unzip_dir} sqlines.log " \
            f"{self.in_file} {self.out_file} {self.log_file} " \
            f'if [ -d {self.file_path} ] ' \
            f'then echo "exists" else echo "not exists" fi'
        self.log.info(cmd2)
        res2 = self.root.sh(cmd2).result()
        self.log.info(res2)
        self.assertIn('Uninstall Sqlines successfully.', res1,
                      'Sqlines卸载失败：' + step_txt)
        self.assertIn('not exists', cmd2, '删除文件失败' + cmd2)
        self.log.info(f'----- {os.path.basename(__file__)} end-----')
